# 配置参数
class Config:
    seed = 42
    device = "cuda:0" if torch.cuda.is_available() else "cpu"
    batch_size = 8
    lr = 0.01
    epochs = 20
    seq_vocab = "AUCG"
    coord_dims = 7  # 7个骨架点
    hidden_dim = 128
    k_neighbors = 5  # 每个节点的近邻数